package modbusGreen

import (
    "encoding/binary"
    "errors"
    "fmt"
    "github.com/tarm/serial"
    "sync"
    "time"
)


var (
    serialConn      *serial.Port
    serialInterval  = 200 * int64(time.Millisecond)
    serialWritelock sync.Mutex //串口写操作锁
    seriallock      sync.Mutex //串口写操作锁
    devStr          string
    // sfjState        = 0x00
    sfjControl = 0x00
    getCode    = 3

    syslog []string
)

// GetDataByReg 通过传感器id获取指定节点数据
func GetDataByReg(id , nodeId int) (data []uint16, err error) {

    // 获取到指定传感器
    register, err := GetRegisterById(id)
    if err != nil {
        fmt.Println("err: ", err)
    }

    // 获取数据
    data, err = GetData(nodeId, getCode, register.RegisterAddr, register.RegisterLength)
    if err != nil {
        fmt.Println("err: ", err)
    }
    return
}

func GetAndSaveRecordByReg(id, nodeId, sensorId int) (err error) {
    data, err := GetDataByReg(id, nodeId)
    if err != nil {
        Error.Println(err)
        return err
    }
    when := time.Now().Unix()
    registerRecord := new(RegisterRecord)
    registerRecord.RegisterId = id
    registerRecord.SensorId = sensorId
    registerRecord.Value = int(data[0])
    registerRecord.Time = when
    registerRecord.Unit = "%"
    err = AddRegisterRecord(registerRecord)
    if err != nil {
        Error.Println(err)
        return err

    }
    return
}

// GetDatasBySensor 通过传感器ID，获取其下所有寄存器数据
func GetDatasBySensor(id int) (datas map[int][]uint16, err error) {
    // 获取该传感器下所有寄存器id
    registers, err := GetRegisterBySensor(id)
    if err != nil {
        return nil, err
    }

    datas = make(map[int][]uint16, len(registers))
    // 查询出所有的数据，并组织数据结构存储
    for _, v := range registers {
        data, err := GetDataByReg(v.Id, 1)
        if err != nil {
            return nil, err
        }
        datas[v.Id] = data
    }
    return
}

// GetData 获取数据
func GetData(nodCode, funcCode, addr, dataLen int) (data []uint16,err error) {

    // generate request frame
    req_frame := generateFrame(nodCode,funcCode,addr,dataLen)

    // write to serial
    err = serialWrite(req_frame)
    if err != nil {
        return nil, errors.New("serialWrite error")
    }

    time.Sleep(30 * time.Millisecond)
    // read from serial
    res, err := serialRead()
    if err != nil {
        return nil, errors.New("serialRead error")
    }
    // 存储有效数据
    ////* 求出有效数据长度
    validDataLen := (len(res)-5) / 2
    data = make([]uint16, validDataLen)
    index := 0

    //从第4个数据到倒数第3个为有效数据
    for i := 3 ; i<len(res)-2; i+= 2 {
        tmp := make([]byte, 2)
        tmp[0] = res[i]
        tmp[1] = res[i+1]

        resUnit16 := binary.BigEndian.Uint16(tmp)
        data[index] = resUnit16
        index++
    }
    //tmp[2] = res[3]
    //tmp[3] = res[4]
    return data, nil
}

// get dataByCount
func GetDataByCount(nodCode, funcCode, addr, dataLen, count int) (data []uint16,err error) {

    // generate request frame
    req_frame := generateFrame(nodCode,funcCode,addr,dataLen)

    // write to serial
    err = serialWrite(req_frame)
    if err != nil {
        return nil, errors.New("serialWrite error")
    }

    time.Sleep(30 * time.Millisecond)
    // read from serial
    res, err := serialReadByCount(count)
    if err != nil {
        return nil, errors.New("serialRead error")
    }
    // 存储有效数据
    ////* 求出有效数据长度
    validDataLen := (len(res)-5) / 2
    data = make([]uint16, validDataLen)
    index := 0

    //从第4个数据到倒数第3个为有效数据
    for i := 3 ; i<len(res)-2; i+= 2 {
        tmp := make([]byte, 2)
        tmp[0] = res[i]
        tmp[1] = res[i+1]

        resUnit16 := binary.BigEndian.Uint16(tmp)
        data[index] = resUnit16
        index++
    }
    //tmp[2] = res[3]
    //tmp[3] = res[4]
    return data, nil
}

// 生成响应码
func generateFrame(nodCode, funcCode, addr, len int) (requestFrame []byte) {

    buf_addr := GenerateLittleEndian(addr)
    buf_len := GenerateLittleEndian(len)

    data := []byte{uint8(nodCode), uint8(funcCode), buf_addr[1], buf_addr[0], buf_len[1], buf_len[0]}
    data = fillChecksum(data)
    requestFrame = data
    return
}

func GenerateLittleEndian(data int) (buf []byte) {
    buf = make([]byte, 2)
    binary.LittleEndian.PutUint16(buf, uint16(data))
    return
}

//FillChecksum 填充校验和
func fillChecksum(data []byte) []byte {
    sum := CheckSum(data)
    b := getIntByte(int(sum))
    data = append(data, b[3])
    data = append(data, b[2])
    return data
}

//CheckSum CheckSum
func CheckSum(data []byte) uint16 {
    var crc16 uint16
    crc16 = 0xffff
    for _, v := range data {
        n := uint8(uint16(v) ^ crc16)
        crc16 >>= 8
        crc16 ^= MbTable[n]
    }
    return crc16
}

//GetIntByte 将Int转byte
func getIntByte(i int) []byte {
    b := make([]byte, 4)
    binary.BigEndian.PutUint32(b, uint32(i))
    return b
}

//MbTable MbTable
var MbTable = []uint16{
    0X0000, 0XC0C1, 0XC181, 0X0140, 0XC301, 0X03C0, 0X0280, 0XC241,
    0XC601, 0X06C0, 0X0780, 0XC741, 0X0500, 0XC5C1, 0XC481, 0X0440,
    0XCC01, 0X0CC0, 0X0D80, 0XCD41, 0X0F00, 0XCFC1, 0XCE81, 0X0E40,
    0X0A00, 0XCAC1, 0XCB81, 0X0B40, 0XC901, 0X09C0, 0X0880, 0XC841,
    0XD801, 0X18C0, 0X1980, 0XD941, 0X1B00, 0XDBC1, 0XDA81, 0X1A40,
    0X1E00, 0XDEC1, 0XDF81, 0X1F40, 0XDD01, 0X1DC0, 0X1C80, 0XDC41,
    0X1400, 0XD4C1, 0XD581, 0X1540, 0XD701, 0X17C0, 0X1680, 0XD641,
    0XD201, 0X12C0, 0X1380, 0XD341, 0X1100, 0XD1C1, 0XD081, 0X1040,
    0XF001, 0X30C0, 0X3180, 0XF141, 0X3300, 0XF3C1, 0XF281, 0X3240,
    0X3600, 0XF6C1, 0XF781, 0X3740, 0XF501, 0X35C0, 0X3480, 0XF441,
    0X3C00, 0XFCC1, 0XFD81, 0X3D40, 0XFF01, 0X3FC0, 0X3E80, 0XFE41,
    0XFA01, 0X3AC0, 0X3B80, 0XFB41, 0X3900, 0XF9C1, 0XF881, 0X3840,
    0X2800, 0XE8C1, 0XE981, 0X2940, 0XEB01, 0X2BC0, 0X2A80, 0XEA41,
    0XEE01, 0X2EC0, 0X2F80, 0XEF41, 0X2D00, 0XEDC1, 0XEC81, 0X2C40,
    0XE401, 0X24C0, 0X2580, 0XE541, 0X2700, 0XE7C1, 0XE681, 0X2640,
    0X2200, 0XE2C1, 0XE381, 0X2340, 0XE101, 0X21C0, 0X2080, 0XE041,
    0XA001, 0X60C0, 0X6180, 0XA141, 0X6300, 0XA3C1, 0XA281, 0X6240,
    0X6600, 0XA6C1, 0XA781, 0X6740, 0XA501, 0X65C0, 0X6480, 0XA441,
    0X6C00, 0XACC1, 0XAD81, 0X6D40, 0XAF01, 0X6FC0, 0X6E80, 0XAE41,
    0XAA01, 0X6AC0, 0X6B80, 0XAB41, 0X6900, 0XA9C1, 0XA881, 0X6840,
    0X7800, 0XB8C1, 0XB981, 0X7940, 0XBB01, 0X7BC0, 0X7A80, 0XBA41,
    0XBE01, 0X7EC0, 0X7F80, 0XBF41, 0X7D00, 0XBDC1, 0XBC81, 0X7C40,
    0XB401, 0X74C0, 0X7580, 0XB541, 0X7700, 0XB7C1, 0XB681, 0X7640,
    0X7200, 0XB2C1, 0XB381, 0X7340, 0XB101, 0X71C0, 0X7080, 0XB041,
    0X5000, 0X90C1, 0X9181, 0X5140, 0X9301, 0X53C0, 0X5280, 0X9241,
    0X9601, 0X56C0, 0X5780, 0X9741, 0X5500, 0X95C1, 0X9481, 0X5440,
    0X9C01, 0X5CC0, 0X5D80, 0X9D41, 0X5F00, 0X9FC1, 0X9E81, 0X5E40,
    0X5A00, 0X9AC1, 0X9B81, 0X5B40, 0X9901, 0X59C0, 0X5880, 0X9841,
    0X8801, 0X48C0, 0X4980, 0X8941, 0X4B00, 0X8BC1, 0X8A81, 0X4A40,
    0X4E00, 0X8EC1, 0X8F81, 0X4F40, 0X8D01, 0X4DC0, 0X4C80, 0X8C41,
    0X4400, 0X84C1, 0X8581, 0X4540, 0X8701, 0X47C0, 0X4680, 0X8641,
    0X8201, 0X42C0, 0X4380, 0X8341, 0X4100, 0X81C1, 0X8081, 0X4040,
}

// 向传感器发送数据
func serialWrite(data []byte) (err error) {
    if serialConn == nil {
        return errors.New("serial error")
    }
    serialWritelock.Lock()
    defer serialWritelock.Unlock()
    _, err = serialConn.Write(data)
    //insertLog(fmt.Sprintf("%s %X", "send:", data))
    return err
}

// 从传感器读取响应数据
func serialRead() (data []byte, err error) {
    if serialConn == nil {
        return nil, errors.New("serial error")
    }
    // 从寄存器读取所有数据
    bufs := make([]byte, 256)
    count, err := serialConn.Read(bufs)
    if err != nil {
        return nil, err
    }

    data = make([]byte, count)
    ti := time.Now().UnixNano()
    for index := 0; index < count; {
        buf := bufs[:count]
        //n, err := serialConn.Read(buf)
        //if err != nil {
        //	return nil, err
        //}
        for Jndex := 0; Jndex < count ; Jndex++ {
            data[index] = buf[Jndex]
            index++
        }
        if index == count {
            ti = time.Now().UnixNano()
        }
        if index != count && time.Now().UnixNano()-ti-serialInterval > 0 {
            // syslog.info("串口延迟统计 ", time.Duration(time.Now().UnixNano()-ti), buf[:n])
            ti = time.Now().UnixNano()
            index = 0
        }
    }
    // SysLog.Info("recv ", fmt.Sprintf("%X", data))
    //insertLog(fmt.Sprintf("%s %X", "recv:", data))

    if !verifyCheckSum(data) {
        serialConn.Flush()
        // syslog.error("CheckSum Error ", data)
        return nil, errors.New("CheckSum Error ")
    }
    return data, nil
}

// serialReadByCount 通过指定字节读取串口数据
func serialReadByCount(count int) (data []byte, err error) {
    if serialConn == nil {
        return nil, errors.New("serial error")
    }
    data = make([]byte, count)
    ti := time.Now().UnixNano()
    for index := 0; index < count; {
        buf := make([]byte, count)
        n, err := serialConn.Read(buf)
        if err != nil {
            return nil, err
        }
        for Jndex := 0; Jndex < n && index < count; Jndex++ {
            data[index] = buf[Jndex]
            index++
        }
        if index == n {
            ti = time.Now().UnixNano()
        }
        if index != count && time.Now().UnixNano()-ti-serialInterval > 0 {
            // syslog.info("串口延迟统计 ", time.Duration(time.Now().UnixNano()-ti), buf[:n])
            ti = time.Now().UnixNano()
            index = 0
        }
    }
    // SysLog.Info("recv ", fmt.Sprintf("%X", data))
    //insertLog(fmt.Sprintf("%s %X", "recv:", data))

    if !verifyCheckSum(data) {
        serialConn.Flush()
        // syslog.error("CheckSum Error ", data)
        return nil, errors.New("CheckSum Error ")
    }
    return data, nil
}

// VerifyCheckSum 验证校验和
func verifyCheckSum(data []byte) bool {
    if len(data) < 2 {
        fmt.Print("hhaha")
        return false
    }
    l := len(data) - 2
    sum := CheckSum(data[:l])
    s := binary.LittleEndian.Uint16(data[l:])
    return s == sum
}

// initSerial 初始化串口
func initSerial(dev string) (err error) {
    conf := &serial.Config{
        Name:        dev,
        Baud:        9600,
        ReadTimeout: time.Second * 5,
    }
    if serialConn != nil {
        serialConn.Close()
        serialConn = nil
    }
    serialConn, err = serial.OpenPort(conf)
    if err != nil {
        return err
    }
    return
}